from tortoise import BaseDBAsyncClient


async def upgrade(db: BaseDBAsyncClient) -> str:
    return """
        ALTER TABLE `menu` ADD `is_menu` BOOL NOT NULL  COMMENT '是否为菜单' DEFAULT 1;"""


async def downgrade(db: BaseDBAsyncClient) -> str:
    return """
        ALTER TABLE `menu` DROP COLUMN `is_menu`;"""
